history = classification(num_train_samples = 16000, 
                         num_validation_samples = 2000, batch_size = 512)
plot(history,epochs=20)
